function proxy
    # 如果有传入参数，直接使用参数作为代理地址
    if set -q argv[1]
        set proxy_address $argv[1]
        echo "使用自定义代理地址: $proxy_address"
        # 检查是否通过SSH远程登录
    else if set -q SSH_CONNECTION
        # 提取SSH客户端IP（第一个字段）
        set ssh_client (string split " " $SSH_CONNECTION)[1]
        set proxy_address "http://$ssh_client:7890" # 使用指定的7890端口
        echo "检测到SSH连接，使用远程地址: $proxy_address"
        # 默认使用本地地址
    else
        set proxy_address "http://127.0.0.1:7890" # 使用指定的7890端口
        echo "使用本地代理地址: $proxy_address"
    end

    # 设置代理环境变量
    set -gx http_proxy $proxy_address
    set -gx https_proxy $proxy_address
    set -gx all_proxy $proxy_address

    echo "代理已设置为: $proxy_address"
end
